package cn.iocoder.idata.module.idata.service.dataset;

import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import org.springframework.validation.annotation.Validated;

import java.util.*;
import cn.iocoder.idata.module.idata.controller.admin.dataset.vo.*;
import cn.iocoder.idata.module.idata.dal.dataobject.dataset.DataSetDO;
import cn.iocoder.idata.framework.common.pojo.PageResult;

import cn.iocoder.idata.module.idata.convert.dataset.DataSetConvert;
import cn.iocoder.idata.module.idata.dal.mysql.dataset.DataSetMapper;

import static cn.iocoder.idata.framework.common.exception.util.ServiceExceptionUtil.exception;
import static cn.iocoder.idata.module.system.enums.ErrorCodeConstants.*;

/**
 * 数据集 Service 实现类
 *
 * @author 智数源码
 */
@Service
@Validated
public class DataSetServiceImpl implements DataSetService {

    @Resource
    private DataSetMapper dataSetMapper;

    @Override
    public Long createDataSet(DataSetCreateReqVO createReqVO) {
        // 插入
        DataSetDO dataSet = DataSetConvert.INSTANCE.convert(createReqVO);
        dataSetMapper.insert(dataSet);
        // 返回
        return dataSet.getId();
    }

    @Override
    public void updateDataSet(DataSetUpdateReqVO updateReqVO) {
        // 校验存在
        validateDataSetExists(updateReqVO.getId());
        // 更新
        DataSetDO updateObj = DataSetConvert.INSTANCE.convert(updateReqVO);
        dataSetMapper.updateById(updateObj);
    }

    @Override
    public void deleteDataSet(Long id) {
        // 校验存在
        validateDataSetExists(id);
        // 删除
        dataSetMapper.deleteById(id);
    }

    private void validateDataSetExists(Long id) {
        if (dataSetMapper.selectById(id) == null) {
            throw exception(DATA_SET_NOT_EXISTS);
        }
    }

    @Override
    public DataSetDO getDataSet(Long id) {
        return dataSetMapper.selectById(id);
    }

    @Override
    public List<DataSetDO> getDataSetList(Collection<Long> ids) {
        return dataSetMapper.selectBatchIds(ids);
    }

    @Override
    public PageResult<DataSetDO> getDataSetPage(DataSetPageReqVO pageReqVO) {
        return dataSetMapper.selectPage(pageReqVO);
    }

    @Override
    public List<DataSetDO> getDataSetList(DataSetExportReqVO exportReqVO) {
        return dataSetMapper.selectList(exportReqVO);
    }

}
